home *** CD-ROM | disk | FTP | other *** search
/ Chip 2007 January, February, March & April / Chip-Cover-CD-2007-02.iso / Pakiet bezpieczenstwa / mini Pentoo LiveCD 2006.1 / mpentoo-2006.1.iso / livecd.squashfs / usr / lib / python2.4 / idlelib / IOBinding.pyc (.txt) < prev    next >
Encoding:
Python Compiled Bytecode  |  2005-10-18  |  15.9 KB  |  615 lines
  1. # Source Generated with Decompyle++
  2. # File: in.pyc (Python 2.4)
  3.  
  4. import os
  5. import types
  6. import sys
  7. import codecs
  8. import tempfile
  9. import tkFileDialog
  10. import tkMessageBox
  11. import re
  12. from Tkinter import *
  13. from SimpleDialog import SimpleDialog
  14. from configHandler import idleConf
  15.  
  16. try:
  17.     from codecs import BOM_UTF8
  18. except ImportError:
  19.     BOM_UTF8 = '\xef\xbb\xbf'
  20.  
  21.  
  22. try:
  23.     import locale
  24.     locale.setlocale(locale.LC_CTYPE, '')
  25. except (ImportError, locale.Error):
  26.     pass
  27.  
  28. encoding = 'ascii'
  29. if sys.platform == 'win32':
  30.     
  31.     try:
  32.         encoding = locale.getdefaultlocale()[1]
  33.         codecs.lookup(encoding)
  34.     except LookupError:
  35.         pass
  36.  
  37. else:
  38.     
  39.     try:
  40.         encoding = locale.nl_langinfo(locale.CODESET)
  41.         if encoding is None or encoding is '':
  42.             encoding = 'ascii'
  43.         
  44.         codecs.lookup(encoding)
  45.     except (NameError, AttributeError, LookupError):
  46.         
  47.         try:
  48.             encoding = locale.getdefaultlocale()[1]
  49.             if encoding is None or encoding is '':
  50.                 encoding = 'ascii'
  51.             
  52.             codecs.lookup(encoding)
  53.         except (ValueError, LookupError):
  54.             pass
  55.  
  56.  
  57. encoding = encoding.lower()
  58. coding_re = re.compile('coding[:=]\\s*([-\\w_.]+)')
  59.  
  60. class EncodingMessage(SimpleDialog):
  61.     '''Inform user that an encoding declaration is needed.'''
  62.     
  63.     def __init__(self, master, enc):
  64.         self.should_edit = False
  65.         self.root = top = Toplevel(master)
  66.         top.bind('<Return>', self.return_event)
  67.         top.bind('<Escape>', self.do_ok)
  68.         top.protocol('WM_DELETE_WINDOW', self.wm_delete_window)
  69.         top.wm_title('I/O Warning')
  70.         top.wm_iconname('I/O Warning')
  71.         self.top = top
  72.         l1 = Label(top, text = 'Non-ASCII found, yet no encoding declared. Add a line like')
  73.         l1.pack(side = TOP, anchor = W)
  74.         l2 = Entry(top, font = 'courier')
  75.         l2.insert(0, '# -*- coding: %s -*-' % enc)
  76.         l2.pack(side = TOP, anchor = W, fill = X)
  77.         l3 = Label(top, text = 'to your file\nChoose OK to save this file as %s\nEdit your general options to silence this warning' % enc)
  78.         l3.pack(side = TOP, anchor = W)
  79.         buttons = Frame(top)
  80.         buttons.pack(side = TOP, fill = X)
  81.         self.default = self.cancel = 0
  82.         b1 = Button(buttons, text = 'Ok', default = 'active', command = self.do_ok)
  83.         b1.pack(side = LEFT, fill = BOTH, expand = 1)
  84.         b2 = Button(buttons, text = 'Edit my file', command = self.do_edit)
  85.         b2.pack(side = LEFT, fill = BOTH, expand = 1)
  86.         self._set_transient(master)
  87.  
  88.     
  89.     def do_ok(self):
  90.         self.done(0)
  91.  
  92.     
  93.     def do_edit(self):
  94.         self.done(1)
  95.  
  96.  
  97.  
  98. def coding_spec(str):
  99.     '''Return the encoding declaration according to PEP 263.
  100.  
  101.     Raise LookupError if the encoding is declared but unknown.
  102.     '''
  103.     str = str.split('\n')[:2]
  104.     str = '\n'.join(str)
  105.     match = coding_re.search(str)
  106.     if not match:
  107.         return None
  108.     
  109.     name = match.group(1)
  110.     import codecs
  111.     
  112.     try:
  113.         codecs.lookup(name)
  114.     except LookupError:
  115.         raise LookupError, 'Unknown encoding ' + name
  116.  
  117.     return name
  118.  
  119.  
  120. class IOBinding:
  121.     
  122.     def __init__(self, editwin):
  123.         self.editwin = editwin
  124.         self.text = editwin.text
  125.         self._IOBinding__id_open = self.text.bind('<<open-window-from-file>>', self.open)
  126.         self._IOBinding__id_save = self.text.bind('<<save-window>>', self.save)
  127.         self._IOBinding__id_saveas = self.text.bind('<<save-window-as-file>>', self.save_as)
  128.         self._IOBinding__id_savecopy = self.text.bind('<<save-copy-of-window-as-file>>', self.save_a_copy)
  129.         self.fileencoding = None
  130.         self._IOBinding__id_print = self.text.bind('<<print-window>>', self.print_window)
  131.  
  132.     
  133.     def close(self):
  134.         self.text.unbind('<<open-window-from-file>>', self._IOBinding__id_open)
  135.         self.text.unbind('<<save-window>>', self._IOBinding__id_save)
  136.         self.text.unbind('<<save-window-as-file>>', self._IOBinding__id_saveas)
  137.         self.text.unbind('<<save-copy-of-window-as-file>>', self._IOBinding__id_savecopy)
  138.         self.text.unbind('<<print-window>>', self._IOBinding__id_print)
  139.         self.editwin = None
  140.         self.text = None
  141.         self.filename_change_hook = None
  142.  
  143.     
  144.     def get_saved(self):
  145.         return self.editwin.get_saved()
  146.  
  147.     
  148.     def set_saved(self, flag):
  149.         self.editwin.set_saved(flag)
  150.  
  151.     
  152.     def reset_undo(self):
  153.         self.editwin.reset_undo()
  154.  
  155.     filename_change_hook = None
  156.     
  157.     def set_filename_change_hook(self, hook):
  158.         self.filename_change_hook = hook
  159.  
  160.     filename = None
  161.     dirname = None
  162.     
  163.     def set_filename(self, filename):
  164.         if filename and os.path.isdir(filename):
  165.             self.filename = None
  166.             self.dirname = filename
  167.         else:
  168.             self.filename = filename
  169.             self.dirname = None
  170.             self.set_saved(1)
  171.             if self.filename_change_hook:
  172.                 self.filename_change_hook()
  173.             
  174.  
  175.     
  176.     def open(self, event = None, editFile = None):
  177.         if self.editwin.flist:
  178.             if not editFile:
  179.                 filename = self.askopenfile()
  180.             else:
  181.                 filename = editFile
  182.             if filename:
  183.                 
  184.                 try:
  185.                     interp = self.editwin.interp
  186.                 except:
  187.                     interp = None
  188.  
  189.                 if not (self.filename) and self.get_saved() and not interp:
  190.                     self.editwin.flist.open(filename, self.loadfile)
  191.                 else:
  192.                     self.editwin.flist.open(filename)
  193.             else:
  194.                 self.text.focus_set()
  195.             return 'break'
  196.         
  197.         if self.get_saved():
  198.             reply = self.maybesave()
  199.             if reply == 'cancel':
  200.                 self.text.focus_set()
  201.                 return 'break'
  202.             
  203.         
  204.         if not editFile:
  205.             filename = self.askopenfile()
  206.         else:
  207.             filename = editFile
  208.         if filename:
  209.             self.loadfile(filename)
  210.         else:
  211.             self.text.focus_set()
  212.         return 'break'
  213.  
  214.     eol = '(\\r\\n)|\\n|\\r'
  215.     eol_re = re.compile(eol)
  216.     eol_convention = os.linesep
  217.     
  218.     def loadfile(self, filename):
  219.         
  220.         try:
  221.             f = open(filename, 'rb')
  222.             chars = f.read()
  223.             f.close()
  224.         except IOError:
  225.             msg = None
  226.             tkMessageBox.showerror('I/O Error', str(msg), master = self.text)
  227.             return False
  228.  
  229.         chars = self.decode(chars)
  230.         firsteol = self.eol_re.search(chars)
  231.         if firsteol:
  232.             self.eol_convention = firsteol.group(0)
  233.             if isinstance(self.eol_convention, unicode):
  234.                 self.eol_convention = self.eol_convention.encode('ascii')
  235.             
  236.             chars = self.eol_re.sub('\\n', chars)
  237.         
  238.         self.text.delete('1.0', 'end')
  239.         self.set_filename(None)
  240.         self.text.insert('1.0', chars)
  241.         self.reset_undo()
  242.         self.set_filename(filename)
  243.         self.text.mark_set('insert', '1.0')
  244.         self.text.see('insert')
  245.         self.updaterecentfileslist(filename)
  246.         return True
  247.  
  248.     
  249.     def decode(self, chars):
  250.         '''Create a Unicode string
  251.  
  252.         If that fails, let Tcl try its best
  253.         '''
  254.         if chars.startswith(BOM_UTF8):
  255.             
  256.             try:
  257.                 chars = chars[3:].decode('utf-8')
  258.             except UnicodeError:
  259.                 return chars
  260.  
  261.             self.fileencoding = BOM_UTF8
  262.             return chars
  263.         
  264.         
  265.         try:
  266.             enc = coding_spec(chars)
  267.         except LookupError:
  268.             name = None
  269.             tkMessageBox.showerror(title = 'Error loading the file', message = "The encoding '%s' is not known to this Python installation. The file may not display correctly" % name, master = self.text)
  270.             enc = None
  271.  
  272.         if enc:
  273.             
  274.             try:
  275.                 return unicode(chars, enc)
  276.             except UnicodeError:
  277.                 pass
  278.             except:
  279.                 None<EXCEPTION MATCH>UnicodeError
  280.             
  281.  
  282.         None<EXCEPTION MATCH>UnicodeError
  283.         
  284.         try:
  285.             return unicode(chars, 'ascii')
  286.         except UnicodeError:
  287.             pass
  288.  
  289.         
  290.         try:
  291.             chars = unicode(chars, encoding)
  292.             self.fileencoding = encoding
  293.         except UnicodeError:
  294.             pass
  295.  
  296.         return chars
  297.  
  298.     
  299.     def maybesave(self):
  300.         if self.get_saved():
  301.             return 'yes'
  302.         
  303.         if not self.filename:
  304.             pass
  305.         message = 'Do you want to save %s before closing?' % 'this untitled document'
  306.         m = tkMessageBox.Message(title = 'Save On Close', message = message, icon = tkMessageBox.QUESTION, type = tkMessageBox.YESNOCANCEL, master = self.text)
  307.         reply = m.show()
  308.         if reply == 'yes':
  309.             self.save(None)
  310.             if not self.get_saved():
  311.                 reply = 'cancel'
  312.             
  313.         
  314.         self.text.focus_set()
  315.         return reply
  316.  
  317.     
  318.     def save(self, event):
  319.         if not self.filename:
  320.             self.save_as(event)
  321.         elif self.writefile(self.filename):
  322.             self.set_saved(1)
  323.             
  324.             try:
  325.                 self.editwin.store_file_breaks()
  326.             except AttributeError:
  327.                 pass
  328.             except:
  329.                 None<EXCEPTION MATCH>AttributeError
  330.             
  331.  
  332.         None<EXCEPTION MATCH>AttributeError
  333.         self.text.focus_set()
  334.         return 'break'
  335.  
  336.     
  337.     def save_as(self, event):
  338.         filename = self.asksavefile()
  339.         if filename:
  340.             if self.writefile(filename):
  341.                 self.set_filename(filename)
  342.                 self.set_saved(1)
  343.                 
  344.                 try:
  345.                     self.editwin.store_file_breaks()
  346.                 except AttributeError:
  347.                     pass
  348.                 except:
  349.                     None<EXCEPTION MATCH>AttributeError
  350.                 
  351.  
  352.             None<EXCEPTION MATCH>AttributeError
  353.         
  354.         self.text.focus_set()
  355.         self.updaterecentfileslist(filename)
  356.         return 'break'
  357.  
  358.     
  359.     def save_a_copy(self, event):
  360.         filename = self.asksavefile()
  361.         if filename:
  362.             self.writefile(filename)
  363.         
  364.         self.text.focus_set()
  365.         self.updaterecentfileslist(filename)
  366.         return 'break'
  367.  
  368.     
  369.     def writefile(self, filename):
  370.         self.fixlastline()
  371.         chars = self.encode(self.text.get('1.0', 'end-1c'))
  372.         if self.eol_convention != '\n':
  373.             chars = chars.replace('\n', self.eol_convention)
  374.         
  375.         
  376.         try:
  377.             f = open(filename, 'wb')
  378.             f.write(chars)
  379.             f.close()
  380.             return True
  381.         except IOError:
  382.             msg = None
  383.             tkMessageBox.showerror('I/O Error', str(msg), master = self.text)
  384.             return False
  385.  
  386.  
  387.     
  388.     def encode(self, chars):
  389.         if isinstance(chars, types.StringType):
  390.             return chars
  391.         
  392.         
  393.         try:
  394.             return chars.encode('ascii')
  395.         except UnicodeError:
  396.             pass
  397.  
  398.         
  399.         try:
  400.             enc = coding_spec(chars)
  401.             failed = None
  402.         except LookupError:
  403.             msg = None
  404.             failed = msg
  405.             enc = None
  406.  
  407.         if enc:
  408.             
  409.             try:
  410.                 return chars.encode(enc)
  411.             except UnicodeError:
  412.                 failed = "Invalid encoding '%s'" % enc
  413.             except:
  414.                 None<EXCEPTION MATCH>UnicodeError
  415.             
  416.  
  417.         None<EXCEPTION MATCH>UnicodeError
  418.         if failed:
  419.             tkMessageBox.showerror('I/O Error', '%s. Saving as UTF-8' % failed, master = self.text)
  420.         
  421.         if self.fileencoding == BOM_UTF8 or failed:
  422.             return BOM_UTF8 + chars.encode('utf-8')
  423.         
  424.         if self.fileencoding:
  425.             
  426.             try:
  427.                 return chars.encode(self.fileencoding)
  428.             except UnicodeError:
  429.                 tkMessageBox.showerror('I/O Error', "Cannot save this as '%s' anymore. Saving as UTF-8" % self.fileencoding, master = self.text)
  430.                 return BOM_UTF8 + chars.encode('utf-8')
  431.             except:
  432.                 None<EXCEPTION MATCH>UnicodeError
  433.             
  434.  
  435.         None<EXCEPTION MATCH>UnicodeError
  436.         config_encoding = idleConf.GetOption('main', 'EditorWindow', 'encoding')
  437.         if config_encoding == 'utf-8':
  438.             return BOM_UTF8 + chars.encode('utf-8')
  439.         
  440.         ask_user = True
  441.         
  442.         try:
  443.             chars = chars.encode(encoding)
  444.             enc = encoding
  445.             if config_encoding == 'locale':
  446.                 ask_user = False
  447.         except UnicodeError:
  448.             chars = BOM_UTF8 + chars.encode('utf-8')
  449.             enc = 'utf-8'
  450.  
  451.         if not ask_user:
  452.             return chars
  453.         
  454.         dialog = EncodingMessage(self.editwin.top, enc)
  455.         dialog.go()
  456.         if dialog.num == 1:
  457.             encline = '# -*- coding: %s -*-\n' % enc
  458.             firstline = self.text.get('1.0', '2.0')
  459.             if firstline.startswith('#!'):
  460.                 self.text.insert('2.0', encline)
  461.             else:
  462.                 self.text.insert('1.0', encline)
  463.             return self.encode(self.text.get('1.0', 'end-1c'))
  464.         
  465.         return chars
  466.  
  467.     
  468.     def fixlastline(self):
  469.         c = self.text.get('end-2c')
  470.         if c != '\n':
  471.             self.text.insert('end-1c', '\n')
  472.         
  473.  
  474.     
  475.     def print_window(self, event):
  476.         tempfilename = None
  477.         saved = self.get_saved()
  478.         if saved:
  479.             filename = self.filename
  480.         
  481.         if not saved or filename is None:
  482.             (tfd, tempfilename) = tempfile.mkstemp(prefix = 'IDLE_tmp_')
  483.             filename = tempfilename
  484.             os.close(tfd)
  485.             if not self.writefile(tempfilename):
  486.                 os.unlink(tempfilename)
  487.                 return 'break'
  488.             
  489.         
  490.         platform = os.name
  491.         printPlatform = 1
  492.         if platform == 'posix':
  493.             command = idleConf.GetOption('main', 'General', 'print-command-posix')
  494.             command = command + ' 2>&1'
  495.         elif platform == 'nt':
  496.             command = idleConf.GetOption('main', 'General', 'print-command-win')
  497.         else:
  498.             printPlatform = 0
  499.         if printPlatform:
  500.             command = command % filename
  501.             pipe = os.popen(command, 'r')
  502.             output = pipe.read().strip()
  503.             status = pipe.close()
  504.             if status:
  505.                 output = 'Printing failed (exit status 0x%x)\n' % status + output
  506.             
  507.             if output:
  508.                 output = 'Printing command: %s\n' % repr(command) + output
  509.                 tkMessageBox.showerror('Print status', output, master = self.text)
  510.             
  511.         else:
  512.             message = 'Printing is not enabled for this platform: %s' % platform
  513.             tkMessageBox.showinfo('Print status', message, master = self.text)
  514.         if tempfilename:
  515.             os.unlink(tempfilename)
  516.         
  517.         return 'break'
  518.  
  519.     opendialog = None
  520.     savedialog = None
  521.     filetypes = [
  522.         ('Python and text files', '*.py *.pyw *.txt', 'TEXT'),
  523.         ('All text files', '*', 'TEXT'),
  524.         ('All files', '*')]
  525.     
  526.     def askopenfile(self):
  527.         (dir, base) = self.defaultfilename('open')
  528.         if not self.opendialog:
  529.             self.opendialog = tkFileDialog.Open(master = self.text, filetypes = self.filetypes)
  530.         
  531.         return self.opendialog.show(initialdir = dir, initialfile = base)
  532.  
  533.     
  534.     def defaultfilename(self, mode = 'open'):
  535.         if self.filename:
  536.             return os.path.split(self.filename)
  537.         elif self.dirname:
  538.             return (self.dirname, '')
  539.         else:
  540.             
  541.             try:
  542.                 pwd = os.getcwd()
  543.             except os.error:
  544.                 pwd = ''
  545.  
  546.             return (pwd, '')
  547.  
  548.     
  549.     def asksavefile(self):
  550.         (dir, base) = self.defaultfilename('save')
  551.         if not self.savedialog:
  552.             self.savedialog = tkFileDialog.SaveAs(master = self.text, filetypes = self.filetypes)
  553.         
  554.         return self.savedialog.show(initialdir = dir, initialfile = base)
  555.  
  556.     
  557.     def updaterecentfileslist(self, filename):
  558.         '''Update recent file list on all editor windows'''
  559.         self.editwin.update_recent_files_list(filename)
  560.  
  561.  
  562.  
  563. def test():
  564.     root = Tk()
  565.     
  566.     class MyEditWin:
  567.         
  568.         def __init__(self, text):
  569.             self.text = text
  570.             self.flist = None
  571.             self.text.bind('<Control-o>', self.open)
  572.             self.text.bind('<Control-s>', self.save)
  573.             self.text.bind('<Alt-s>', self.save_as)
  574.             self.text.bind('<Alt-z>', self.save_a_copy)
  575.  
  576.         
  577.         def get_saved(self):
  578.             return 0
  579.  
  580.         
  581.         def set_saved(self, flag):
  582.             pass
  583.  
  584.         
  585.         def reset_undo(self):
  586.             pass
  587.  
  588.         
  589.         def open(self, event):
  590.             self.text.event_generate('<<open-window-from-file>>')
  591.  
  592.         
  593.         def save(self, event):
  594.             self.text.event_generate('<<save-window>>')
  595.  
  596.         
  597.         def save_as(self, event):
  598.             self.text.event_generate('<<save-window-as-file>>')
  599.  
  600.         
  601.         def save_a_copy(self, event):
  602.             self.text.event_generate('<<save-copy-of-window-as-file>>')
  603.  
  604.  
  605.     text = Text(root)
  606.     text.pack()
  607.     text.focus_set()
  608.     editwin = MyEditWin(text)
  609.     io = IOBinding(editwin)
  610.     root.mainloop()
  611.  
  612. if __name__ == '__main__':
  613.     test()
  614.  
  615.